package oj;

import java.util.HashMap;
import java.util.Map;

class Node2 {
    int val;
    Node2 next;
    Node2 random;

    public Node2(int val) {
        this.val = val;
        this.next = null;
        this.random = null;
    }
}

public class LC138 {
    public Node2 copyRandomList(Node2 head) {
        if(head==null) return null;
        //创建一个哈希表，key是原节点，value是新节点
        Map<Node2, Node2> nodeMap=new HashMap<>();
        //将原节点和新节点放入哈希表中
        for(Node2 x = head; x!=null; x=x.next){
            Node2 newNode=new Node2(x.val);
            nodeMap.put(x,newNode);
        }
        //遍历原链表，设置新节点的next和random
        for(Node2 x = head; x!=null; x=x.next){
            //map.get(x)是对应的新节点，x.next是原节点的下一个
            //map.get(p.next)是原节点下一个对应的新节点
            nodeMap.get(x).next=nodeMap.get(x.next);
            //p.random是原节点随机指向
            //map.get(p.random)是原节点随机指向  对应的新节点
            nodeMap.get(x).random=nodeMap.get(x.random);
        }
        return nodeMap.get(head);
    }
}
